Skip to content

Comments

remove torchvision dependency#1178

Merged
jeffra merged 1 commit intomasterfrom
jeffra-patch-2
Jun 21, 2021
Merged

remove torchvision dependency#1178
jeffra merged 1 commit intomasterfrom
jeffra-patch-2

Conversation

@jeffra
Copy link
Collaborator

@jeffra jeffra commented Jun 21, 2021

DeepSpeed itself does not require torchvision, I think we originally added it to support our cifar example. However we should just have this requirement in the example code itself and not in base deepspeed. We ran into strange issues with torchvision and compiled versions of torch not being compatible, and thus made deepspeed trigger version mismatch errors.

@jeffra jeffra merged commit 3b68984 into master Jun 21, 2021
@jeffra jeffra deleted the jeffra-patch-2 branch June 21, 2021 17:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants